package models

import (
	"fmt"
	"log"

	"ix-brow/config"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

// 全局数据库连接
var DB *gorm.DB

// 初始化数据库连接和模型迁移
func InitDB() error {
	// 构建DSN
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		config.AppConfig.Database.User,
		config.AppConfig.Database.Password,
		config.AppConfig.Database.Host,
		config.AppConfig.Database.Port,
		config.AppConfig.Database.DBName)

	// 配置GORM
	gormConfig := &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	}

	// 连接数据库
	var err error
	DB, err = gorm.Open(mysql.Open(dsn), gormConfig)
	if err != nil {
		return fmt.Errorf("数据库连接失败: %v", err)
	}

	log.Println("数据库连接成功")

	// 迁移数据库表结构
	log.Println("开始数据库迁移...")

	if err := DB.AutoMigrate(&User{}, &Group{}, &Browser{}, &Proxy{}, &ProxyApplication{}); err != nil {
		return fmt.Errorf("数据库迁移失败: %v", err)
	}

	log.Println("数据库迁移完成")
	return nil
}
